package cn.edu.zut.pms.faker;

import com.github.javafaker.Faker;
import java.sql.*;
import java.util.Locale;
import java.util.Random;

public class FakerOrderRegion {
    public static void main(String[] args) {
        // 数据库连接信息（请根据实际环境修改）
        String url = "jdbc:mysql://192.168.17.150:3306/soil?allowPublicKeyRetrieval=true&useSSL=false";
        String user = "root";
        String password = "000000";

        // 创建中文Faker实例
        Faker faker = new Faker(new Locale("zh_CN"));
        Random random = new Random();

        // 定义销售区域（与源表dwd_logistics_data保持一致）
        String[] salesRegions = {"华北区", "华东区", "华南区", "西北区", "西南区", "东北区", "华中区"};
        // 区域权重（模拟实际业务中各区域订单分布差异）
        int[] regionWeights = {15, 20, 18, 10, 12, 10, 15}; // 总和为100

        try (Connection connection = DriverManager.getConnection(url, user, password)) {
            // 插入数据的SQL语句
            String insertSql = "INSERT INTO dwd_order_region (" +
                    "order_id, sales_region" +
                    ") VALUES (?, ?)";

            try (PreparedStatement preparedStatement = connection.prepareStatement(insertSql)) {
                // 生成400条有效关联的订单区域记录（可调整数量）
                for (int i = 0; i < 400; i++) {
                    // 生成与源表关联的有效订单号（格式与dwd_logistics_data、dwd_order_base保持一致）
                    String orderId = "ORD" + faker.number().numberBetween(10000000, 99999999);

                    // 按权重随机选择销售区域（华东、华南等经济发达区域订单更多）
                    String salesRegion = getRegionByWeight(salesRegions, regionWeights, random);

                    // 设置参数（确保order_id非空，符合表设计要求）
                    preparedStatement.setString(1, orderId);
                    preparedStatement.setString(2, salesRegion);

                    preparedStatement.executeUpdate();
                }
                System.out.println("数据插入完成，共插入 400 条订单区域信息记录");
            } catch (SQLException e) {
                e.printStackTrace();
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    // 按权重随机选择区域（权重越高，被选中概率越大）
    private static String getRegionByWeight(String[] regions, int[] weights, Random random) {
        int total = 0;
        for (int weight : weights) {
            total += weight;
        }
        int rand = random.nextInt(total);
        int sum = 0;
        for (int i = 0; i < regions.length; i++) {
            sum += weights[i];
            if (rand < sum) {
                return regions[i];
            }
        }
        return regions[0]; // 兜底返回第一个区域
    }
}